package cn.itsource.global;

import cn.itsource.global.exception.BusinessException;
import cn.itsource.base.util.AjaxResult;
import cn.itsource.base.util.ErrorCode;
import org.springframework.validation.ObjectError;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import java.util.List;
import java.util.stream.Collectors;

/**
 * 全局异常处理器
 *    去监听你的controller有没有报错
 */
@RestControllerAdvice //复合注解：RestController类的增强  - 可以指定包
public class GlobalExceptionHandler {

    /**
     * ExceptionHandler  专门用来处理指定异常的
     * @param e
     * @return
     */
    @ExceptionHandler(BusinessException.class)
    public AjaxResult globleException(BusinessException e){
//        e.printStackTrace();//打印错误信息
//        return AjaxResult.me().setSuccess(false).setMessage(e.getMessage());
        String code = e.getMessage();
        ErrorCode errorCode = ErrorCode.getErrorCoudeByCode(code);
        return AjaxResult.me().setErrorCode(errorCode);
    }

    @ExceptionHandler(MethodArgumentNotValidException.class)//去指定要捕获的那个异常
    public AjaxResult methodArgumentNotValidException(MethodArgumentNotValidException e){
        List<ObjectError> allErrors = e.getBindingResult().getAllErrors();
        List<String> collect = allErrors.stream().map(ObjectError::getDefaultMessage ).collect(Collectors.toList());
        String join = String.join(",", collect);//将list里面所有的数据用指定的格式进行字符串拼接
        return AjaxResult.me().setSuccess(false).setMessage(join);

    }
    @ExceptionHandler(ArithmeticException.class)
    public AjaxResult arithmeticException(ArithmeticException e){
        e.printStackTrace();//打印错误信息
        return AjaxResult.me().setErrorCode(ErrorCode.ARITHMETIC_ERROR);
    }


    //捕获非业务异常
    //捕捉其他异常
    @ExceptionHandler(Exception.class)
    public AjaxResult exception(Exception e){
        e.printStackTrace();
        return AjaxResult.me().setSuccess(false).setMessage("系统内部异常");
    }
}
